﻿// Copyright(c) Microsoft Corporation
// All rights reserved.
//
// Licensed under the Apache License, Version 2.0 (the License); you may not use
// this file except in compliance with the License. You may obtain a copy of the
// License at http://www.apache.org/licenses/LICENSE-2.0
//
// THIS CODE IS PROVIDED ON AN  *AS IS* BASIS, WITHOUT WARRANTIES OR CONDITIONS
// OF ANY KIND, EITHER EXPRESS OR IMPLIED, INCLUDING WITHOUT LIMITATION ANY
// IMPLIED WARRANTIES OR CONDITIONS OF TITLE, FITNESS FOR A PARTICULAR PURPOSE,
// MERCHANTABILITY OR NON-INFRINGEMENT.
//
// See the Apache Version 2.0 License for specific language governing
// permissions and limitations under the License.

using Microsoft.Python.Analysis.Diagnostics;
using Microsoft.Python.LanguageServer.Protocol;
using Microsoft.Python.Parsing;

namespace Microsoft.Python.LanguageServer.Diagnostics {
    internal static class DiagnosticExtensions {
        public static Diagnostic ToDiagnostic(this DiagnosticsEntry diagnostic, string source = "Python") {
            return new Diagnostic {
                range = diagnostic.SourceSpan,
                severity = diagnostic.Severity.ToDiagnosticSeverity(),
                source = source,
                code = diagnostic.ErrorCode,
                message = diagnostic.Message,
            };
        }

        public static DiagnosticSeverity ToDiagnosticSeverity(this Severity severity) {
            switch (severity) {
                case Severity.Warning:
                    return DiagnosticSeverity.Warning;
                case Severity.Information:
                    return DiagnosticSeverity.Information;
                case Severity.Hint:
                    return DiagnosticSeverity.Hint;
                default:
                    return DiagnosticSeverity.Error;
            }
        }
    }
}
